import torch as t
from torch import Tensor as T
from torch.autograd import Variable as V


class Perceptron(t.nn.Module):
    def __init__(self, in_feather, hidden_feather, out_feather):
        super(Perceptron, self).__init__()
        self.layer1 = t.nn.Linear(in_feather, hidden_feather)

        self.layer2 = t.nn.Linear(hidden_feather, out_feather)

    def forward(self, x):
        x = self.layer1(x)
        x = t.sigmoid(x)
        return self.layer2(x)


perceptron = Perceptron(3, 4, 1)
for name, param in perceptron.named_parameters():
    print(name, param)
